簡介:在課程中我們將使用最前沿的JAX框架,帶你來感受體驗最為真實的深度學習的算法開發過程。課程內容從算法原理到算法實戰,從思路到代碼,從框架到具體模型編碼,助力你輕松入門前沿框架JAX。
第1章 JAX基礎概覽
與常見深度學習進行對比,并介紹了JAX的起源、特點和基礎案例。從本章開始帶領同學使用JAX解決基礎的建模問題,并為之后的內容提供知識鋪墊。
第2章 JAX極速入門
講解了JAX的安裝方法和使用流程,通過房價預測和圖像分類帶領同學們學會使用JAX解決實際的問題,包括損失函數計算、梯度計算和參數更新的細節。
第3章 JAX搭建服飾分類模型
在JAX基礎知識點的基礎上,深入JAX的實現過程。本章將通過JAX搭建深度學習模型,從數據集讀取到模型搭建,講解JAX模型搭建的細節流程。
第4章 JAX訓練服飾分類模型
在服飾分類模型的基礎上,講解JAX損失函數、訓練流程、模型保存和預測。最終帶領同學完成深度學習模型的訓練流程,完成實際模型的訓練和調優。